# -*- coding: utf-8 -*-
"""Setup the Suporte-Pais-Python application"""

import logging

import transaction
from tg import config

from suportepaispython.config.environment import load_environment

__all__ = ['setup_app']

log = logging.getLogger(__name__)


def setup_app(command, conf, vars):
    """Place any commands to setup suportepaispython here"""
    load_environment(conf.global_conf, conf.local_conf)
    # Load the models
    from suportepaispython import model
    print "Creating tables"
    model.metadata.create_all(bind=config['pylons.app_globals'].sa_engine)

    pais0 = model.Pais("BR","Brasil","br","abc")
    pais1 = model.Pais("MEX","México","mex","abc")
    pais2 = model.Pais("CAN","Canadá","can","aaa")
    unidadefederativa0 = model.UnidadeFederativa("RJ","Rio de janeiro")
    unidadefederativa1 = model.UnidadeFederativa("SP","Sao Paulo")
    pais0.unidadeFederativaCollection.append(unidadefederativa0)
    pais0.unidadeFederativaCollection.append(unidadefederativa1)
    
    
    model.DBSession.add(pais0)
    model.DBSession.add(pais1)
    model.DBSession.add(pais2)
    model.DBSession.add(unidadefederativa0)
    model.DBSession.add(unidadefederativa1)

    transaction.commit()
    print "Successfully setup"
